# -*- coding: utf-8 -*-

"""
@Datetime: 2019/5/21
@Author: Zhang Yafei
"""


def func_arg(arg):
    def func(functionName):
        def func_in():
            print("---记录日志--arg=%s----" % arg)
            if arg == "heihei":
                functionName()
                functionName()
            else:
                functionName()

        return func_in

    return func


# 1.先执行func_arg("heihei")函数，这个函数return 的结果是func这个函数的引用
# 2.@func
# 3使用@func对test进行装饰
@func_arg("heihei")
def test():
    print("---test----")


@func_arg("haha")
def test2():
    print("-----test2------")


test()
test2()
